package com.darren.center.shardingjdbc.demo.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

import java.util.Map;

@Component
@Slf4j
public class ApplicationContextSupport implements ApplicationContextAware{

    private static ApplicationContext applicationContext;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        if(ApplicationContextSupport.applicationContext == null) {
            ApplicationContextSupport.applicationContext = applicationContext;
        }
    }

    //获取applicationContext
    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }

    //通过name获取 Bean.
    public static Object getBean(String name){
        return getApplicationContext().getBean(name);
    }

    //通过class获取Bean.
    public static <T> T getBean(Class<T> clazz){
        return getApplicationContext().getBean(clazz);
    }

    //通过name,以及Clazz返回指定的Bean
    public static <T> T getBean(String name,Class<T> clazz){
        return getApplicationContext().getBean(name, clazz);
    }

    public static <T> Map<String, T> getBeansByClass(Class<T> c) {
        return !isInit() ? null : applicationContext.getBeansOfType(c);
    }

    public static boolean isInit() {
        if (null == applicationContext) {
            log.error("###### SpringContextUtil没有成功初始化，请检查Spring上下文自动扫描包是否包含它所在的包");
            return false;
        } else {
            return true;
        }
    }


}
